package com.xs.aifitnessagent.service;

import com.baomidou.mybatisplus.extension.service.IService;
import com.xs.aifitnessagent.entities.Message;
import org.springframework.ai.chat.messages.MessageType;

import java.util.List;

/**
 * 消息服务接口
 *
 * @author lambor
 */
public interface MessageService extends IService<Message> {

    /**
     * 保存消息（统一方法）
     *
     * @param chatId 聊天ID
     * @param userId 用户ID
     * @param content 消息内容
     * @param messageType 消息类型：USER用户消息，ASSISTANT助手消息
     * @param modelName 模型名称（仅助手消息需要，用户消息传null）
     * @return 保存的消息
     */
    Message saveMessage(String chatId, String userId, String content, String messageType, String modelName);

    /**
     * 根据聊天ID获取消息列表
     *
     * @param chatId 聊天ID
     * @param userId 用户ID
     * @return 消息列表
     */
    List<Message> getMessagesByChatId(String chatId, String userId);

    /**
     * 编辑用户消息
     *
     * @param messageId 消息ID
     * @param newContent 新内容
     * @param userId 用户ID
     * @return 更新后的消息
     */
    Message editUserMessage(Long messageId, String newContent, String userId);

    /**
     * 删除聊天的所有消息
     *
     * @param chatId 聊天ID
     * @param userId 用户ID
     */
    void deleteMessagesByChatId(String chatId, String userId);

    /**
     * 删除指定消息之后的所有消息
     *
     * @param messageId 消息ID
     * @param chatId 聊天ID
     * @param userId 用户ID
     * @return 删除的消息数量
     */
    int deleteMessagesAfter(Long messageId, String chatId, String userId);
}